package info.wangyuan.agent.service.rag;

import info.wangyuan.agent.common.Enum.RagDocSyncOcrEnum;
import jakarta.annotation.Resource;
import org.springframework.stereotype.Service;

import java.util.Map;

/**
 * 异步切分上下文
 *
 * @author Albert
 * @since 2025-08-30 21:44:31
 */
@Service
public class RagDocSyncChunkContext {

    @Resource
    private Map<String, RagDocSyncChunkStrategy> taskExportStrategyMap;

    public RagDocSyncChunkStrategy getTaskExportStrategy(String strategy) {
        return taskExportStrategyMap.get(RagDocSyncOcrEnum.getLabelByValue(strategy));
    }
}
